Chained assignment in Python